# coding:utf-8
import os
import shutil
from conftest import BASE_DIR

# #执行整个框架前，如果report目录下有上次的报告文件，先删除
# def test_delect_report():
#     save_path = BASE_DIR + r'\report'
#     if os.listdir(save_path):
#         file_list = os.listdir(save_path)
#         for f in file_list:
#             file_path = os.path.join(save_path, f)
#             if os.path.isfile(file_path):
#                 os.remove(file_path)
#             elif os.path.isdir(file_path):
#                 shutil.rmtree(file_path, True)

#执行整个框架前，如果logger目录下有上次的日志文件，先删除
"""
这个函数会遍历指定目录下的所有文件，如果文件名不是__init__.py，它就会删除这个文件。注意，这个函数不会删除子目录，无论它们是否为空"""

def test_delect_log():
    directory = BASE_DIR + r'/logger'
    [os.remove(os.path.join(directory, f)) for f in os.listdir(directory) if f != '__init__.py']

def test_delect_error():
    # save_path = BASE_DIR + r'\error'
    # if os.listdir(save_path):
    #     file_list = os.listdir(save_path)
    #     for f in file_list:
    #         file_path = os.path.join(save_path, f)
    #         if os.path.isfile(file_path):
    #             os.remove(file_path)
    #         elif os.path.isdir(file_path):
    #             shutil.rmtree(file_path, True)
    directory = BASE_DIR + r'/error'
    [os.remove(os.path.join(directory, f)) for f in os.listdir(directory) if f != '__init__.py']


def test_delect_scripts_report():
    path=BASE_DIR + r'\scripts\report'
    # 不存在就创建目录
    if not os.path.exists(path):
        pass
    else:  # 否则就删除目录下的文件
        for f in os.listdir(path):
            file_path = os.path.join(path, f)  # 拼接文件路径
            os.remove(file_path)  # 删除文件（如果 file_path 是一个文件夹，将抛出OSError）
        os.rmdir(path)  # 删除文件夹 （如果目录 pic_path 非空，则抛出一个OSError异常）


#test_delect_report()
test_delect_log()
test_delect_error()
#test_delect_scripts_report()